import java.util.LinkedList;
import java.util.Queue;

class Solution03 {
    // 数轴表示
    int[] dx = { 0, 0, -1, 1 };
    int[] dy = { 1, -1, 0, 0 };

    public int[][] floodFill(int[][] image, int sr, int sc, int color) {
        // 处理边界条件
        int prev = image[sr][sc];
        if (prev == color) {
            return image;
        }
        int m = image.length, n = image[0].length;

        // 设置一个队列 里面存放数组，用 LinkedList 增加删除效率更高
        Queue<int[]> q = new LinkedList<>();
        // 将起始位置添加到队列里面
        q.add(new int[] { sr, sc });
        // 进行循环条件判断
        while (!q.isEmpty()) {
            // 取出队头元素
            int[] tmp = q.poll();
            int a = tmp[0], b = tmp[1];
            // 进行像素修改
            image[a][b] = color;
            for (int i = 0; i < 4; i++) {
                int x = a + dx[i], y = b + dy[i];
                // 进行边界值判断
                if (x >= 0 && x < m && y >= 0 && y < n && image[x][y] == prev) {
                    // 添加到队列
                    q.add(new int[] { x, y });
                }
            }
        }
        return image;
    }
}